Goto

Collaborating Authors

 turbo autoencoder


Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

Neural Information Processing Systems

Designing codes that combat the noise in a communication medium has remained a significant area of research in information theory as well as wireless communications. Asymptotically optimal channel codes have been developed by mathematicians for communicating under canonical models after over 60 years of research. On the other hand, in many non-canonical channel settings, optimal codes do not exist and the codes designed for canonical models are adapted via heuristics to these channels and are thus not guaranteed to be optimal. In this work, we make significant progress on this problem by designing a fully end-to-end jointly trained neural encoder and decoder, namely, Turbo Autoencoder (TurboAE), with the following contributions: (a) under moderate block lengths, TurboAE approaches state-of-the-art performance under canonical channels; (b) moreover, TurboAE outperforms the state-of-the-art codes under non-canonical settings in terms of reliability. TurboAE shows that the development of channel coding design can be automated via deep learning, with near-optimal performance.


Reviews: Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

Neural Information Processing Systems

In recent years, several papers have employed deep learning methods to decode various classes of codes (turbo codes, linear codes, polar codes). This work focuses on turbo codes, and has the more ambitious goal of providing joint training of the decoder and the encoder (which means that the resulting code will not be a turbo code in the traditional sense). The authors borrow some ideas from the turbo coding literature (e.g., interleaving) and use CNNs to design decoder and encoder (as opposed to RNNs used in several other papers). The proposed TurboAE algorithm achieves performance which is comparable to state-of-the-art codes (see Figure 1). This is quite impressive, even though the code length is quite short (i.e. 100 bits).


Reviews: Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

Neural Information Processing Systems

This paper considers a combination of encoder and decoder architecture, which is a serially-concatenated code with interleavers, as in the turbo codes, combined with turbo-like iterative decoding, and proposes implementing encoders and decoders of the constituent codes with 1D-CNN, which allow us to train the encoders and the decoders in an end-to-end and data-driven fashion. Two reviewers raised concern about the scalability issue of the proposal, and the authors admit in their rebuttal that it is a central question. Although the review scores exhibited a large split in the initial round of review, mainly due to the scalability issue as well as comparison in performance with other existing coding schemes, after the authors' rebuttal all the reviewers rated this paper above the acceptance threshold. I would therefore like to recommend acceptance of this paper.


Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

Neural Information Processing Systems

Designing codes that combat the noise in a communication medium has remained a significant area of research in information theory as well as wireless communications. Asymptotically optimal channel codes have been developed by mathematicians for communicating under canonical models after over 60 years of research. On the other hand, in many non-canonical channel settings, optimal codes do not exist and the codes designed for canonical models are adapted via heuristics to these channels and are thus not guaranteed to be optimal. In this work, we make significant progress on this problem by designing a fully end-to-end jointly trained neural encoder and decoder, namely, Turbo Autoencoder (TurboAE), with the following contributions: (a) under moderate block lengths, TurboAE approaches state-of-the-art performance under canonical channels; (b) moreover, TurboAE outperforms the state-of-the-art codes under non-canonical settings in terms of reliability. TurboAE shows that the development of channel coding design can be automated via deep learning, with near-optimal performance.


Component Training of Turbo Autoencoders

arXiv.org Artificial Intelligence

Isolated training with Gaussian priors (TGP) of the component autoencoders of turbo-autoencoder architectures enables faster, more consistent training and better generalization to arbitrary decoding iterations than training based on deep unfolding. We propose fitting the components via extrinsic information transfer (EXIT) charts to a desired behavior which enables scaling to larger message lengths ($k \approx 1000$) while retaining competitive performance. To the best of our knowledge, this is the first autoencoder that performs close to classical codes in this regime. Although the binary cross-entropy (BCE) loss function optimizes the bit error rate (BER) of the components, the design via EXIT charts enables to focus on the block error rate (BLER). In serially concatenated systems the component-wise TGP approach is well known for inner components with a fixed outer binary interface, e.g., a learned inner code or equalizer, with an outer binary error correcting code. In this paper we extend the component training to structures with an inner and outer autoencoder, where we propose a new 1-bit quantization strategy for the encoder outputs based on the underlying communication problem. Finally, we discuss the model complexity of the learned components during design time (training) and inference and show that the number of weights in the encoder can be reduced by 99.96 %.


Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

Neural Information Processing Systems

Designing codes that combat the noise in a communication medium has remained a significant area of research in information theory as well as wireless communications. Asymptotically optimal channel codes have been developed by mathematicians for communicating under canonical models after over 60 years of research. On the other hand, in many non-canonical channel settings, optimal codes do not exist and the codes designed for canonical models are adapted via heuristics to these channels and are thus not guaranteed to be optimal. In this work, we make significant progress on this problem by designing a fully end-to-end jointly trained neural encoder and decoder, namely, Turbo Autoencoder (TurboAE), with the following contributions: (a) under moderate block lengths, TurboAE approaches state-of-the-art performance under canonical channels; (b) moreover, TurboAE outperforms the state-of-the-art codes under non-canonical settings in terms of reliability. TurboAE shows that the development of channel coding design can be automated via deep learning, with near-optimal performance.